#ifndef _TYPES_H_
#define _TYPES_H_

typedef struct Parameter
{
    std::vector<int> head_shape;
    std::vector<int> input_shape;
    std::vector<int> img_shape;
    std::vector<int> anchor;
    float conf_thres = 0.5;
    float nms_thres = 0.3;
    int num_anchor = 3;
    int class_head = 80;
}Parameter;

#endif